package com.github.yangyishe;

import java.util.Arrays;
import java.util.Stack;

/**
 * 739. 每日温度
 * https://leetcode.cn/problems/daily-temperatures/?envType=study-plan-v2&envId=top-100-liked
 * 此题实际算法与官方算法一致，但不止为何速度要慢很多
 *
 * 给定一个整数数组 temperatures ，表示每天的温度，返回一个数组 answer ，其中 answer[i] 是指对于第 i 天，下一个更高温度出现在几天后。如果气温在这之后都不会升高，请在该位置用 0 来代替。
 *
 *
 *
 * 示例 1:
 *
 * 输入: temperatures = [73,74,75,71,69,72,76,73]
 * 输出: [1,1,4,2,1,1,0,0]
 * 示例 2:
 *
 * 输入: temperatures = [30,40,50,60]
 * 输出: [1,1,1,0]
 * 示例 3:
 *
 * 输入: temperatures = [30,60,90]
 * 输出: [1,1,0]
 *
 *
 * 提示：
 *
 * 1 <= temperatures.length <= 105
 * 30 <= temperatures[i] <= 100
 */
public class Problem739 {
    public static void main(String[] args) {
        int[] temperatures=new int[]{73,74,75,71,69,72,76,73};

        Problem739 problem739 = new Problem739();
        int[] dailyTemperatures = problem739.dailyTemperatures(temperatures);
        System.out.println("Arrays.toString(dailyTemperatures) = " + Arrays.toString(dailyTemperatures));
    }

    /**
     * 思路：
     * 暴力解决法：每一个数字，都从第二个开始，逐个查询最大值
     * 信息的浪费：遍历数字时，浪费了对数字比较的信息收集
     *
     * 利用单调栈，从后向前进行比较即可
     * 解决左侧或右侧，第一个比当前元素大或小的元素
     *
     * @param temperatures
     * @return
     */
    public int[] dailyTemperatures(int[] temperatures) {
        // 寻找右侧第一个比当前元素大的元素
        Stack<Integer> stack=new Stack<>();

        int[] result=new int[temperatures.length];

        for(int fast=temperatures.length-1;fast>=0;fast--){
            while(!stack.isEmpty()&&temperatures[fast]>=temperatures[stack.peek()]){
                stack.pop();
            }
            if(stack.isEmpty()){
                result[fast]=0;
            }else{
                result[fast]=stack.peek()-fast;
            }
            stack.push(fast);
        }

        return result;
    }
}
